from app import create_app, db
from app.models import VaccinationSite

def insert_vaccination_sites():
    # 创建应用上下文
    app = create_app()
    with app.app_context():
        # 先清空现有数据
        VaccinationSite.query.delete()
        
        # 准备数据
        sites_data = [
            {
                'name': '市第一人民医院',
                'address': '市中心路123号',
                'vaccineCount': 200,
                'status': '可预约'
            },
            {
                'name': '市第二人民医院',
                'address': '东城区456号',
                'vaccineCount': 150,
                'status': '可预约'
            },
            {
                'name': '社区卫生服务中心',
                'address': '西城区789号',
                'vaccineCount': 80,
                'status': '已约满'
            },
            {
                'name': '社区卫生服务中心',
                'address': '东城区689号',
                'vaccineCount': 89,
                'status': '可预约'
            },
            {
                'name': '省中和医院',
                'address': '中和新区122号',
                'vaccineCount': 120,
                'status': '可预约'
            },
            {
                'name': '市第一警察人民医院',
                'address': '市中心警察街道101号',
                'vaccineCount': 110,
                'status': '可预约'
            },
            {
                'name': '安康医院',
                'address': '新都区安康街道91号',
                'vaccineCount': 99,
                'status': '可预约'
            }
        ]
        
        # 插入数据
        for site_data in sites_data:
            site = VaccinationSite(**site_data)
            db.session.add(site)
        
        # 提交事务
        try:
            db.session.commit()
            print("数据插入成功！")
        except Exception as e:
            db.session.rollback()
            print(f"数据插入失败：{str(e)}")

if __name__ == '__main__':
    insert_vaccination_sites() 